from starlette.endpoints import HTTPEndpoint
from starlette.responses import JSONResponse
from model.ModEquipVenderProRep import EquipVenderProRep, equipVenderProRepSchema
from model.ModEquipVenderProRepBypro import EquipVenderProRepBypro, equipVenderProRepByproSchema
from database import Session, engine
import datetime


def get_equip_vender_pro(session, bureauid: str, classifyid: str, pro_classifyid: str):
    # 厂家问题分类统计中的明细表
    equip_pro = session.query(EquipVenderProRep).filter(EquipVenderProRep.bureauid == bureauid).filter(
        EquipVenderProRep.classifyid == classifyid).filter(
        EquipVenderProRep.pro_classifyid == pro_classifyid).all()
    return JSONResponse(equipVenderProRepSchema.dump(equip_pro, many=True).data)


def get_equip_pro(session, bureauid: str, classifyid: str):
    # 厂家问题分类统计表
    equip_pro = session.query(EquipVenderProRepBypro).filter(EquipVenderProRepBypro.bureauid == bureauid).filter(
        EquipVenderProRepBypro.classifyid == classifyid).all()
    return JSONResponse(equipVenderProRepByproSchema.dump(equip_pro, many=True).data)
